/** @file
 *****************************************************************************
 Implementation of bigint wrapper class around GMP's MPZ long integers.
 *****************************************************************************
 * @author     This file is part of libff, developed by SCIPR Lab
 *             and contributors (see AUTHORS).
 * @copyright  MIT license (see LICENSE file)
 *****************************************************************************/

#ifndef BIGINT_TCC_
#define BIGINT_TCC_
#include <cassert>
#include <cstring>
#include <random>

namespace libff
{

template<mp_size_t n> bigint<n>::bigint(const unsigned long x)
{
    assert(8 * sizeof(x) <= GMP_NUMB_BITS);
    this->data[0] = x;
}

template<mp_size_t n> bigint<n>::bigint(const char *s)
{
    size_t l = strlen(s);
    unsigned char *s_copy = new unsigned char[l];

    for (size_t i = 0; i < l; ++i) {
        assert(s[i] >= '0' && s[i] <= '9');
        s_copy[i] = s[i] - '0';
    }

    mp_size_t limbs_written = mpn_set_str(this->data, s_copy, l, 10);
    assert(limbs_written <= n);
    UNUSED(limbs_written);

    delete[] s_copy;
}

template<mp_size_t n> bigint<n>::bigint(const mpz_t r)
{
    mpz_t k;
    mpz_init_set(k, r);

    for (size_t i = 0; i < n; ++i) {
        data[i] = mpz_get_ui(k);
        mpz_fdiv_q_2exp(k, k, GMP_NUMB_BITS);
    }

    assert(mpz_sgn(k) == 0);
    mpz_clear(k);
}

template<mp_size_t n> void bigint<n>::print() const
{
    gmp_printf("%Nd\n", this->data, n);
}

template<mp_size_t n> void bigint<n>::print_hex() const
{
    gmp_printf("%Nx\n", this->data, n);
}

template<mp_size_t n> bool bigint<n>::operator==(const bigint<n> &other) const
{
    return (mpn_cmp(this->data, other.data, n) == 0);
}

template<mp_size_t n> bool bigint<n>::operator!=(const bigint<n> &other) const
{
    return !(operator==(other));
}

template<mp_size_t n> void bigint<n>::clear() { mpn_zero(this->data, n); }

template<mp_size_t n> bool bigint<n>::is_zero() const
{
    for (size_t i = 0; i < n; ++i) {
        if (this->data[i]) {
            return false;
        }
    }

    return true;
}

template<mp_size_t n> size_t bigint<n>::num_bits() const
{
    /*
       for (long i = max_bits(); i >= 0; --i) {
           if (this->test_bit(i)) {
               return i+1;
           }
       }

       return 0;
    */
    for (long i = n - 1; i >= 0; --i) {
        mp_limb_t x = this->data[i];
        if (x == 0) {
            continue;
        } else {
            return ((i + 1) * GMP_NUMB_BITS) - __builtin_clzl(x);
        }
    }
    return 0;
}

template<mp_size_t n> unsigned long bigint<n>::as_ulong() const
{
    return this->data[0];
}

template<mp_size_t n> void bigint<n>::to_mpz(mpz_t r) const
{
    mpz_set_ui(r, 0);

    for (int i = n - 1; i >= 0; --i) {
        mpz_mul_2exp(r, r, GMP_NUMB_BITS);
        mpz_add_ui(r, r, this->data[i]);
    }
}

template<mp_size_t n> bool bigint<n>::test_bit(const std::size_t bitno) const
{
    if (bitno >= n * GMP_NUMB_BITS) {
        return false;
    } else {
        const std::size_t part = bitno / GMP_NUMB_BITS;
        const std::size_t bit = bitno - (GMP_NUMB_BITS * part);
        const mp_limb_t one = 1;
        return (this->data[part] & (one << bit)) != 0;
    }
}

template<mp_size_t n> bigint<n> &bigint<n>::randomize()
{
    static_assert(
        GMP_NUMB_BITS == sizeof(mp_limb_t) * 8, "Wrong GMP_NUMB_BITS value");
    std::random_device rd;
    constexpr size_t num_random_words =
        sizeof(mp_limb_t) * n / sizeof(std::random_device::result_type);
    auto random_words =
        reinterpret_cast<std::random_device::result_type *>(this->data);
    for (size_t i = 0; i < num_random_words; ++i) {
        random_words[i] = rd();
    }

    return (*this);
}

template<mp_size_t n>
std::ostream &operator<<(std::ostream &out, const bigint<n> &b)
{
#ifdef BINARY_OUTPUT
    out.write((char *)b.data, sizeof(b.data[0]) * n);
#else
    mpz_t t;
    mpz_init(t);
    b.to_mpz(t);

    out << t;

    mpz_clear(t);
#endif
    return out;
}

template<mp_size_t n> std::istream &operator>>(std::istream &in, bigint<n> &b)
{
#ifdef BINARY_OUTPUT
    in.read((char *)b.data, sizeof(b.data[0]) * n);
#else
    std::string s;
    in >> s;

    size_t l = s.size();
    unsigned char *s_copy = new unsigned char[l];

    for (size_t i = 0; i < l; ++i) {
        assert(s[i] >= '0' && s[i] <= '9');
        s_copy[i] = s[i] - '0';
    }

    mp_size_t limbs_written = mpn_set_str(b.data, s_copy, l, 10);
    assert(limbs_written <= n);
    UNUSED(limbs_written);

    delete[] s_copy;
#endif
    return in;
}

} // namespace libff
#endif // BIGINT_TCC_
